# -*- coding: utf-8 -*-
# @Time    : 2018/3/31 19:14
# @Author  : Tianchiyue
# @File    : configs.py
# @Software: PyCharm Community Edition

rnn_config = {
    'num_classes': 3,
    'dropout_rate': 0.2,
    'rnn_output_size': 300,
    'embed_trainable': True,
    'batch_size': 50,
    'optimizer': 'adam',
    'lr': 0.001,
    'epochs': 20,
    'n_stop': 5,
    'hidden_dims': 300,
    'bidirectional': True,
    'rnn': 'lstm',
    'activation': 'tanh',
    'use_mlp': True,
    'use_l2': False,
    'l2': 0.0001,
    'lr_decay_epoch': 20,
    'lr_decay_rate': 0.5,
    'han': True
}
cnn_config = {
    'filters': 250,
    'kernel_sizes':[3,4,5],
    'num_classes': 3,
    'dropout_rate': 0.2,
    'embed_trainable': True,
    'batch_size': 50,
    'optimizer': 'adam',
    'lr': 0.001,
    'epochs': 20,
    'n_stop': 5,
    'hidden_dims': 250,
    'activation': 'tanh',
    'use_mlp': True,
    'use_l2': False,
    'l2': 0.0001,
    'lr_decay_epoch': 20,
    'lr_decay_rate': 0.5
}


"""
调参结果:
lr,dropout,batch_size
0.001,0.2,50
0.001,0.2,100
TODO dropout为0
     hidden_dims 300
     
"""
ata_config = {
    'use_target':True,
    'num_classes':3,
    'dropout_rate':0.2,
    'lstm_output_size':300,
    'embed_trainable':True,
    'batch_size':50,
    'optimizer':'adam',
    'lr':0.001,
    'epochs':20,
    'n_stop':20,
    'hidden_dims':300,
    'activation': 'tanh',
    'use_mlp': True,
    'use_l2': False,
    'l2': 0.0001,
    'lr_decay_epoch': 20,
    'lr_decay_rate': 0.5
}

tan_config = {
    'use_target': True,
    'num_classes': 3,
    'dropout_rate': 0.0,
    'lstm_output_size': 100,
    'embed_trainable': True,
    'batch_size': 50,
    'optimizer': 'adam',
    'lr': 0.0005,
    'epochs': 10,
    'n_stop': 20,
    'hidden_dims': 100,
    'activation': 'tanh',
    'use_mlp': False,
    'use_l2': True,
    'l2': 0.0001,
    'lr_decay_epoch': 20,
    'lr_decay_rate': 0.5
}